Adding jQuery code to handle the UI stuff for steps