Consider using sympy common subexpression elimination to make returned expressions less long and repetitive. See: https://docs.sympy.org/latest/modules/rewriting.html