added more tests for better coverage