Developed and validated a key feature for the TeamCOMPAS/COMPAS repository, focusing on improving mass sampling accuracy in the Max Moe sampler. The work ensured that the mass distribution of single and binary components strictly followed the Kroupa initial mass function, matching the COSMIC code implementation after applying the same correction. This involved Python scripting, numerical sampling, and rigorous cross-code validation to guarantee identical results between COMPAS and COSMIC. Additional efforts included cleaning up code comments and enhancing readability in sampleMoeDiStefano.py. The work demonstrated skills in Python, data analysis, and scientific computing, emphasizing simulation reliability and code maintainability.