Rework Begin/End calls for shader state restoration
Comment fixes