Improve docs
Improve docs
Some better documentation