FIX some errors

MODIFIED main.py  -  fix some errors