Add in controller, model & shell code