Next: write tests (testing working)