added more to probability calculation