edit the main function