initial version
fixed legend bug

in `plot.train`, a single legend is created and the legend title is
fixed