Python script to read out matrix elements
Matrix elements working