adding balance feature partially