Added shake to main program