Built data structure within train. Need to build out prediction.