Release v0.2.1

* Fix documentation issue