Did some more work with the database and models.py