We're losing the print statements, could we add logging?