Add the ability to set instance_type