still working on the calculating balance feature