change main function