Handling more edge cases.