Added some tests

To the same file as before, but added tests.