Update base.py example