Plotting wavefunctions on band diagram