writing functions in r simplify code and improve efficiency