Add functionality for magnetisation.
Also allow to save configurations and plot them.