Added remove stage option