Fix code examples from review comments