plot.imshow(np.reshape(images[index], (16,16), order='F'), cmap=cm.Greys_r)

def TileImage(imgs, picturesPerRow=16):
    # Convert to a true list of 16x16 images
    tmp = np.reshape(imgs, (-1, 16, 16), order='F')
    img = ""
    for i in range(0, tmp.shape[0], picturesPerRow):
        # On the last iteration, we may not have exactly picturesPerRow
        # images left so we need to pad
        if tmp.shape[0] - i >= picturesPerRow:
            mid = np.concatenate(tmp[i:i+picturesPerRow], axis=1)
        else:
            padding = np.zeros((picturesPerRow - (tmp.shape[0] -i), 16, 16))
            mid = np.concatenate(np.concatenate((tmp[i:tmp.shape[0]], padding), axis=0), axis=1)

        if img == "":
            img = mid
        else:
            img = np.concatenate((img, mid), axis=0)

    return img