Save figure if --plot requested

Maybe not perfect behaviour, but I don't want a million flags...