Add test cases based on the examples