Added the shake functionality