add chapter2 main.py