Not really, Sorry! But…
Adding begin and end.. with a comment on the begin creates regions which would look like this…bit of hack though!
Otherwise you can only expand and collapse you just can’t dictate what should be expanded and collapsed. Not without a third-party tool such as SSMS Tools Pack.