add magnetization stuff