Update models.py

CHG: Added filter by id
Merge pull request #2 from damianpv/patch-2

Update models.py