Add StepController Trait