Bug fix in p_e_given_f_baseline.py