Finite Elements Method implemented in Python