renamed test.py to main.py