Added models and db