Fix no screen shake

shakeAmount was being reset every frame