add os import to main.py