Add test B problem A