made some small changes to the balance section