Better function highlighting