improve demo/mass_uncertainty: better plot