`FUNCTION` not highlighted

[mnaser]

Hi there!

I've noticed that FUNCTION is not highlighted since it's a recent addition, would be nice if we can tack it in there.

Thanks!