current basic version working for prediction