adding code to predict and generate submission file