Update ps4-3-subway_planning.py