main function

Change the main function