<SYSTEM_TASK:>
Solve the following problem using Python, implementing the functions described below, one line at a time
<END_TASK>
<USER_TASK:>
Description:
 def downstream(self, step_name):
 """Returns the direct dependencies of the given step"""

        return list(self.steps[dep] for dep in self.dag.downstream(step_name))