Added Model class to repo