Added some basic test cases