Use args.document_dir