Fixed code block that was spilling out