Dodan generator dreves + popravki

Popravljena optimizacija: sedaj dela pravilno (ali je še linearen čas? -
verjetno ne)