Redone the update(element) method