Update colorizer.py