Updates base.py