Add test coverage
Merge pull request #5 from colinwren/master

Add test coverage