An improved affinity model to enhance variable operating strategy for pumps used as turbines