Change the update method
Working!