Made some small edits and added assets to the controller class.