Change screen shake to linear diminishing

Refactor screen shake to own module