Updated v1.0 with colorization option