Removed the print statements from the test code.