Finish update drive functionality