Made an improvement to the "Virtual Dataset" demo