Fixing package
Updating docs