Went back to running demo on full dataset