Adds missing coloring for some functions.