Revise mass balance

This pull request addresses issue #165.