Add tests to the code that's worth testing. There's not much there now, but it's worth setting up a framework.