Change the Update Function