Improve update system