Add missing code blocks