Fixed few potential errors, fixes #5
Bump version v0.2.0