Add probability of keeping into CUDA version