added more tests to improve coverage