Change main function of test