Adding update functionality.