Try making a function to change models