Add editing for stages