ColorGenerator class after refactor