It's working! But the probability equation is kind of messy.