Fixed code for urban and rural stratification as well as spatial stratification, as well as some checks to make sure the code works even if the total number of households to sample is fewer than the number of households per rural/urban pixels.