how to write a function in r