Added methods to update balance