Add styles for responsive docs

You can use `removeClass` directly, so you can drop these `if`s.