changed umbrella sampling code