Add option to relax magnetisation before time integration