Add some more edge cases

geddit
edge cases