Change a print to logger
Use argparse
Update README