Put argparse into a seperate function to clean up the code.