Tests working for v0.2.0