Adding a few more tests for coverage