Update models.py

add some models