Updated to use the update checker