Add in update checker