Make update system more modular and customizable