Added argparse but has some bugs
Added argparse but has some bugs