Add test for colorize and fix it behavior