Adding set_dependency method to DAG class