Add update checking