lab2: change to argparse