Update end_screen and headless_client modules