Added test for code block