Modifid StepImplementation class to add new steps