Retrieved ModelGeneration and Model directly instead of iterating  models