Drop unnecessary prints
Fix the label assignment order
update xtrain_with_labels