from sympy import *
from sympy.utilities.codegen import codegen
init_printing(use_latex="mathjax")