moved argparse import to the top