added missing self's in models.py