Move argparse import to the function scope