changes on prediction section