Add missing code block