update the testing code