Consolidated UIStage behaviour into main Stage class.