Improving docs

Improving docs