check_balance function

Checks the mass balance of the flows involved