Fixed a couple things in models.py