made some small improvements on the example dataset