Minor bug fixes (missing tree label, missing edge)