Updated the Montgomery dataset and associated tests to fit the new data handling design implemented in the add-datamodule branch