Added Shake functionality