add prediction
adjust to the new events