Fixed bug in last commit. Bumped to version v0.1.1