Remove debug, handle edge cases.