add --stage support