Added debug message for model training