Using 2 classes instead of 4

[valentina-paniza]

Hi! This is a really nice project. I'm trying to execute the pipeline with the trained models, but I want to use 2 classes. I have added that argument on step 4 (python crossvalidate_blending.py --n_classes 2) but I'm wondering how should I add this on the last step (5) where the file submission.py is used. I don't see that "n_classes" is an option, and I don't see either if there's any reference to the previous step.
Thanks in advance!