minor clean up in colorize method