working on prediction