added squeeze to imshow so that X.shape == (N, M, 1) ndarrays will
not raise errors.