Python problem randomization example