Added Abdullah's main2.py