trying to solve prediction