case 2: II.1.1: halfway through, need to fix dim in test set