Further cleanup of featureresponses.py and fixes to plotting code