Add sanity checks for balance