Add color mode, as well as an if == __main__ check.