Feature/tidying up tests

Hey! Thanks for this nice library, it saved me time! :)

I took a look at the test looking for documentation, and took the liberty of tidying them up.