Add a struct to hold the balances