Added colorize function