If the extension could provide hints for separating each stage into sections, it would improve readability without the need for users to explicitly include comments separating the stages.

Slidev offers a feature similar to this.
https://sli.dev/features/vscode-extension

before	after