Work on one-click update functionality