Add option to use colors