Fix #4 bug
Version bump: 0.1.1