Restore colorize function