Add support for es6 arrow functions.

Do you think we can have es6 arrow functions support this way? I tested on my
codebase and it works well. If you are willing to accept this pull request, I'd
test cases for this functionality.