lab5: change to argparse