Create thermalization.py