Run only from main.py