Add test for div