So close! There's still a bug in the probability calculation though.