Keep working that update method to include WishList