Add the controller
Merge pull request #191 from VitSalis/fix_color_bug

Add the controller