corrections in main2.py