updating some of the testing code