Add some cooments to test.py