added prediction and made many of the features one-hot vectors