show machine name in zsh prompt if connected through ssh

.zshrc

function precmd {
        local SSH_STATUS=""

        if [ -n "$SSH_CONNECTION" ]
        then
                SSH_STATUS="%m "
        fi

        PROMPT="${SSH_STATUS}%1~ %B%#%b "
}