Updated docs
Fixed failing test