restructure the testing code