add the second method of test.py