Improve highlighting of javascript function definition