Add tests, clean up code