Added more unit tests
Merge pull request #9 from StephenHynes7/master

Added Unit Test