writing functions in r