Simplify the update files logic