Updating docs and small bug fix