Getting the linearized matrix elements should be easier than it currently is. Preferably it should be done as a function call inside a python script.