Expanded the supported functions and unified the colorization function.