Updated check for colorization