add new file test.py
do some modifications for test.py