Added missing test cases