Close #2 by adding a test thingy