added test.py + fixed calls