Add test code block