Updated docs
Fixed broken test