Account for new parameter in SDE plot