Added css and better handling of the tree