Add tests

Test all the things!