Add argparse module and allow the database to be specified in the arguments.