Moved argparse import to main