changed main.py in lg2 branch