updated readme + a few changes in main.py