Update __init__.py

minor bug fix and optimization for colorwheel (red)