umbrella sampling code