Feynman diagrams with matplotlib