Fix function definition to still highlight 'func'

The change to highlight the name for the function definition was breaking highlighting of the 'func' keyword. This fixes that.