balance tool actually calculates the balance now.
exposed extract_input_address()