Added ability to make a custom plot, some more rearrangements. The main thing left is an ability to meaningfully save the images you see. The only way I know how to do that is to make a separate matplotlib plot and save it, or save each plot individually.