I often put comments that apply to code blocks inside the block at the top, it avoids visual weirdness like this.  Just a suggestion though, this is fine too.