Added more code to the augmented digraph module