now using argparse, separated plotting
now also doing video frames
added re-fitted model to video