Band structure & comparison plot python script