add output for probability calculation