Added some tests!

:shipit: