added creation of vector for prediction/optimization + changed example in main :)