Add train and test datasets