Improved the test coverage
Test fix
Merge pull request #11 from ryanshawty/improved_test_coverage

Improved the test coverage