Added some simple test cases