modified test cases to cover all functions