fixed the plot, added a .h5 save