Add edge case tests