Update likelihood.py