added models and association