Edit: base.py