Move loading of schema to standalone function, only show submit when next stage exists