Update docs
Bug fix