Updated docs
Fixed comments