refactoring the colorize() - done - now just finish the class