Update source_base.py