Add basic test cases