Addede code for shading