Fix major bug and release v0.0.2