for training and testeing