Add option to disable color