Added U(1) gauge heatbath from Alexei.