Add highlighting of user-defined functions

Since it's possible to define user functions with `@function`, they should also be highlighted.