gen plots fixes
pngs of predictions