Changes to main.py and the README file