More implementation for stage control