Adding couple of test cases