Updated missing data handling and dataset compatibility