It works!

All of the tests are condensed down into one function, which makes
things much easier.