Update models_base.py