Read_dataset seems to be working, change write example to save 3D
dataset