Fixed up main.py
Minor changes to readme.