Add edge cases