Added a print test. Ideally would want to test the stdout output but not sure how to.