added models and database