Fix inconsistent colours when some data is missing
Fix issues with plotting mumag4 data
Fix initial derivatives of TR when using relax()