Added Model, Repository and updated controller