Predict works with one X (merged train and test)