worked on prediction