Add optional callback to `pollBalance`