Update motion.jax