Add tests, yay!

Everything is better with tests.