Adding support for arrow functions.