fix bugs - note dimension of b01