The best way to edit a code segment (the "standard" one or the SAS code one) is to position your cursor in the code block, then click the toolbar button (to bring up the separate window again). Add your lines in that window, then close. They will be retained.
Editing the code block "in place" doesn't capture the formatting properly.
@RW9 - I'd like to hear more about your experience with the SAS code display. We have the ability to change that a little bit (formatting/appearance/colored keywords), so your feedback is important.