Adds pyplot to init python packages

docs/12_band_structures_in_higher_dimensions.md

 ```python tags=["initialize"]
 import numpy as np
 import plotly.graph_objs as go
-
-pi = np.pi
+from matplotlib import pyplot as plt 
+from math import pi
 ```
 
 # Tight binding and nearly free electrons