Add missing code and bump version to 0.1.1