Replaced arguments with argparse