Updated to take 3 models