Syntax highlighting works for functions