adding the query and predictions again