Better support for terrain shaders?

Just been trying out terrain shaders and copied relevant sections into the shader as mentioned in the WIKI. These are deleted as soon as you modify and compile the shader.

Would be nice to have a workaround, such as a tick box (or boxes), which would add or keep these with the shader when saving.