Follow-up from "Sample Loaders able to handle certain type of annotations..."

The following discussion from !81 (merged) should be addressed:

I need to do some minor clean-up on MEDS dataset.