Merge pull request #16 from TheMasster12/master

Add unit tests to increase coverage.

Ref #1.