Fix step issue, actually add the step class to the div, doh!