#compdef rlogin rsh ssh

if [[ CURRENT -eq 2 ]]; then
  compgen -k hosts
elif [[ CURRENT -eq 3 ]]; then
  compadd - -l
else
  compgen -u
fi