I've been looking for a similar solution but haven't found one yet. It's frustrating that the comments get collapsed along with the code.