Added a rainbow generator to the controller