Add shake animation