separated calls for train and test