Add support to highlight function names

With this snippet function names will be highlighted in a different
color, note that this does not refer to the function definition i. e.
`def foo()` but instead to a piece of code that calls `foo()`.