Hello, first off thanks for sharing the code for this project! I am interested in trying to use it on a new dataset for a new downstream task and I was wondering if the code is set up to accommodate that currently? I noticed that some of the example datasets are set up explicitly and I wasn't sure if this meant it was currently only set up to run on those datasets. If it can run on other datasets, could you provide some guidance on how to prepare the data for input into the model? I am specifically interested in the ADNI dataset, which I saw you used in the paper but I didn't see reference to it in the code. Again, any guidance would be appreciated. I think this is a really cool and exciting tool. Thanks!