BUG: update example

I took n_leaves out of the constructors for now