Starting to make it handle multiple models.