bug fixes
changes in main.py