After much fiddling, finally got plot_data to plot correctly. Also, added color bar! Wheeeeee\!