Change main function