Fixing colors, see #29

Fixing colors, see #29