Updated main.py *test