Jax main trying to make something different