Case 1 working but with no cross-validation yet.