Final update to FEA solver