Added classes to model, controller and view