Added prints in test.py