Fix selection drawing after refresh. Also attempt to re-select selected node (not yet working).