lab1: change to argparse
lab1: change to argparse