Add option to turn off colorization