added models to models.py