Added models for DB