Comment and spacing update, use print function in tests.py