Refactor - Add get balance infrastructure through websockets