Fixes type annotation for partial
Fixes the type annotation for wrap
Bumps to 0.2, with docs.