Added class to div wrapper