modify ch1.py