Added comment/description
Bug fix Added a return for hosts dict