Move BaseModel to base.py