Also, as subsets tend to mess up things sometimes, pls double check that everything here works fine when this is used for classification later on :)