cleaned up print statements in tests