Update main2.py