Simple python/matplotlib code to make neutrino oscillation plots