Add ability to shake harder at end