Add shake functionality