Improve documentation

Improve documentation