Add a check for save function in editor