fixed minor issues for v0.1.1