Return black/white color buttons logic and abstract it into class category