Corrected a bug in the output train R script