small simplification of models.py