Update colorization.m