Work on select function.
Remove unused styling
Add selected node styling