added a missing call to plt.figure()
fixed plotting for mean field (labels weren't being assigned)