To edit the code in the Code node, do not use the “Edit window” modal window, but maximize the existing container with the code to fit the screen. In this case, you need to narrow the “Input” and “Output” areas as much as possible and hide unnecessary “Mode”, “Language” areas and “Parameters/Settings” tabs, but with the ability to stretch the width of the “Input” and “Output” areas from the minimum to the required width and back, if necessary. You also need to leave the node name (so as not to forget or confuse where the change is made - similar to the name of the procedures) and be sure to leave the “Execute step” button. This will allow you to immediately run the code for execution using the “Execute step” button without unnecessary mode switching without leaving the convenient editing mode. When exiting the editing mode, you just need to return the previous sizes of the areas and show the hidden elements.
In the image I tried to show how it might look:
