Update Shake function