Fix docs
Fix docs, release 0.4.0.1