Add ThermalAtom class