added demo and small data correction