Added more tests for the library

This increases the test coverage.
Ref #406