Need to fix models.py and debug from there