Updated test cases for increased coverage