Quick change to allow class attribute on div tag