add jstree function and some styling