fixed probability calculation -> first working version