Added print statements for classification step