Added some tests

Basically set it up so that I can setup tests more painlessly in the
future.