Finite Element Method Code Python Trai