Added models to the app