Changed the feature vectors to 2D arrays.