The function draw_networkx_edges in the NetworkX python package didn't seem to support arrows. Here is my attempt at implementing an arrow system.