[IMP] adapt models.py