add colorize

Colorize result, refactoring