Update to models.py - database logic 

P.S. Natalie, you have not left my heart yet!