Added tests and improved output text
Merge pull request #9 from ryanshawty/additional_tests

Added tests and improved output text