working main.py file, enjoy