Improve documentation and example