Added step statement to save function