fixed up the axis labeling etc, a bit, and added config for other datasets