Fix bug in import, add legend to plot.

The experiment is done.