Adding test for the custom divs