Adding gen level information to trees