Update NeuralDataClass.py