made a short script to test a probability question