Write a couple of test cases