What I really want is to plot contribution form particular atom in device dos, however in 2D Brillouin zone as function of kx and ky at E=EF. I was essentially thinking if there is anyway of inserting elements information line 38 i.e, parallel_calculator = NLEngine.ParallelDensityOfStatesCalculator(stdv_energies, stdv_kpoints of above script kdos.py.