Robustness In Machine Learning Models: Addressing Distribution Shifts In Medical Imaging