test.py working