Write more tests!

