Add tests. Yay for tests. Testing is good.