Added a model to make it a MVC