Update Jax.lua