Update main.C

Dev modify