Change screen function to return iterator of promise results