Update models.py | corrected typing mistakes & self