Added -f/--file option to plot.py for easy saving.

This option also stops the plot from popping up.