completing models.py