lab4: change to argparse