Correcting tests
Ready for 0.1.0