Thermal Equilibrium Abundances (TEA) code