Achieve at least 95% coverage with unit tests.

[pieterjansevanrensburg]

To ensure that the code is working as intended, it needs to be tested. The first stage to achieving this will be to test the code by making use of the unittest python module.