Add test for multiple code blocks