Update models.py

add related name