Update Tree.cs, sjetih se buga