Updated some minor details in main.