Update ch04_ex03_extended.py