Simple changes in main.