Question about the thermal thin_wall

Hello everyone,
I've been exploring this code with a keen interest in solving thermoplastic problems. However, I've encountered a stumbling block while trying to work with the thin_wall. I'm facing a ModuleNotFoundError: as detailed below:
`
ModuleNotFoundError Traceback (most recent call last)
Cell In[1], line 10
7 import time
9 from jax_fem.generate_mesh import Mesh
---> 10 from jax_fem.core import FEM
11 from jax_fem.solver import solver
12 from jax_fem.utils import save_sol

ModuleNotFoundError: No module named 'jax_fem.core'
`
But I can't find the core file in jax-fem. I'm wondering if there's something I might miss. Any guidance or suggestions would be greatly appreciated!