Adding the update implementation