making => work with function highlighting