main.py additions