Feature new stage ui

added back in. thanks for the catch