Added additional test cases