added more tests for better test coverage