Finite Element Method in Python