Added parameter for new model