added a couple test cases