Commented out print statements, added argparse to handle command line arguments