How to get the simulated pressure variable along the trajectory? For example in this notebook: https://github.com/google/jax-cfd/blob/main/notebooks/demo.ipynb Thank you very much!