Updated colorize function

Just a tiny enhancement