updated Driver.py to calculate P(Z0)