Changed Z-only thermostat option to apply to either Langevin or Rescaling