<SYSTEM_TASK:>
Solve the following problem using Python, implementing the functions described below, one line at a time
<END_TASK>
<USER_TASK:>
Description:
 def draw(self):
 """Draws the tree. Args: ages (array):
 Contains the ages you want to draw. """

        for age, level in enumerate(self.tree.get_branches()):
            if age in self.ages:
                thickness = self._get_thickness(age)
                color = self._get_color(age)
                for branch in level:
                    self._draw_branch(branch, color, thickness, age)