add shake for prediction