A lot of improvements to GUI and plot method.