Update main.C