Added a couple test cases.