Logger instead of print statements?