Fixed model typo added code to controller