Added return statement to main function of test.