add the rest of the stage work