Added some test cases for our use cases