Create Train and store in pickle.py