Update models.py

Import base models