Improved user experience with argparse, so the parameters can be easily set in the command line.