update main2.py