changes to sample prediction