Good point, it will be cleaner to have a function to load a HDF5 and read data and label from it. From line 51 until 58