Add further edge cases