Update models.py

Just some cleaning...