Cleanup, argparse, documentation