Added models and UI