plot updates correctly now. added argparse for plotting