added models and DB