accept 1D data

[laurentperrinet]

• test with shape = (1, 256)
• make the plot function more intelligent