Refactor - add the AddressModel and BalanceSocket functionality