Fixes main_script.py
Fixes #1