Started adding support for colorization

NOT WORKING