Remove some redundancy in flows.py.
Make augmented_matrix.py quieter.