This is will be an example code written specifically for the thermalisation examples.