updated example to new dataset