Updated for working code. Need to add smaller sample data.