Added highlighting for functions

New version of the request #61