Added more prints in test.py