Change data to train.data in order to make it more explicit