Add max flow tests