Think of more edge cases and unit test them to ensure the code is correct.

Unit tests on their own will never give us 100% confidence, but they will be a good way to catch low hanging bugs. For more thorough testing, we will move also write more integration tests to be soak tested in a separate task.