Modified tests to use main.py