Updated tests.
Up to 0.1.1 because tests