Add tests for specific cases