created feature vector function on the train lolz