Add probabiliy distribution class