Update setup function