Update models.py

Added foreign key relationships to other classes