Add tests for edge cases