Code clean up
Added animation feeling to hide and show
Hide/show child nodes and the connection line