update
add test.py, and test