Add more update functionality.