Refactoring to use documentElement instead of body

Also using classlist as we're now advocating using polyfills