Code refactor, disable_colorization feature