Removed separate UI for stager, merged behaviour into main class.