Add stage 4 puzzle, stage 3 minor change