flake8 the file
add argparse