Add shake functionality
Add shake functionality