Minor update to include get_hosts()