Update Step Builder to work with strings

[becega]

RIght now step builder needs to be updated every time we add step, we need for it to work with the class names.