Add balance back in