Improve documentation of the stage parameter