Improve documentation in script