Fix a bug and add the skeleton for keep's unit test

Bumped the version to 0.2.1 after this small bug fix. I'll update the pypi version as soon as this is merged.