Fix color output
Merge pull request #13 from creynders/fix_colors

Fix color output