Add test for code_block