Changing the treemenu script. Didnt like the previous one