Added better gui, TODO full working JTree