add type (train / test)