Refactored color-me, added tests.