Including index for matplotlib notebook