Implement __contains__ method
Add more test cases
Improve the test cases
Organize testcases neatly