added models and code