Add unit tests

Issue #3 resolved
Richard Cotton repo owner created an issue

No description provided.