Update models.py

CHG: Added filter by id