Got drawing of selected nodes working properly