Create some more test coverage for the project.

[Scoombe]

Need some test coverage for the components of the project.

This needs to cover some of the major functions.

Any improvements welcome.

[jakewanders]

I will put in a PR with some additional tests.