Added print statements to inference