Improve docs
Update docs