restored main.py to spec