Make unit tests

Now that you more or less have your funcionality ready and are doing refactoring, it's a good idea to write tests that check all is well so you know when an update breaks something!