Update Main_2d.py