Update models.py

Changed the Database models