Updating class to className on DOM elements