Add some tests. #18. @Luey you should be able to add some as well now it's set up