Small plotting fixes, updating docs to 0.2.1, and git tag!