Improved docs

Improved docs