fixing main.py and updating readme