Added highlighting for functions