Modifying to use a harmonic bath instead of Lammps