diff --git a/video_prediction_tools/main_scripts/main_preprocess_data_step2.py b/video_prediction_tools/main_scripts/main_preprocess_data_step2.py
index 1582ff2d0671e8d3806307fdd6288b97e6a1e441..4010e67d97c5a6a4d4f758f5d51a519213bf5957 100644
--- a/video_prediction_tools/main_scripts/main_preprocess_data_step2.py
+++ b/video_prediction_tools/main_scripts/main_preprocess_data_step2.py
@@ -1,5 +1,24 @@
-
-
+import argparse
+import sys
+import os
+import glob
+import itertools
+import pickle
+import random
+import re
+import numpy as np
+import json
+import tensorflow as tf
+from collections import OrderedDict
+from tensorflow.contrib.training import HParams
+from mpi4py import MPI
+from video_prediction.datasets.base_dataset import VarLenFeatureVideoDataset
+import data_preproces.process_netCDF_v2
+from general_utils import get_unique_vars
+from statistics import Calc_data_stat
+from metadata import MetaData
+from normalization import Norm_data
+from video_prediction.datasets import era5_dataset
 
 
 def main():