added models and migration