Added shake function

!!