Added a few test cases