added models for database