Updated test for lab D2