moved some to main.py