New feature: Added some basic test cases