Use 'with' to automagically call output.finish()
Use ARGS convention for argparse.
Update docstring example.