Running final. Still trying to tweak Probability code