Some tweaks to colorization