added more to the models