Use argparse for plotting memory usage.
Additions to test script for testing and illustrating usage.