Add function to save screenshot and function to get the names of the collision links