Use argparse to pass in weight from the command line.