Update code block for ssh key step