Added a couple more test cases