Add a couple of test cases.