Rework wait_for_results and fill in documentation