update the plotting code plotfits.py.