fix errors in models.py