Added Controller & Model class