Adding more functionality to update scripts.