Make classifier functional for stable systems. Also edit datagen.