Added arguments to the script with module argparse