fix models.py bugs