correcting main2.py