added probability map