Add some test.py