removed the print statements and added a test
2 files changed