Add final code block