- started implementing generating train and test data