Improve probability map