Add balance calculations