updated the models.py to include all models. Need to be implemented entirely