refactor colorization