Optimize B.flatten()