basics of probability in python