DAWNSCI-903 fix dataset to be plotted

 * ensure dimensions of data are correct - cannot be > 1
 * clean up import