Add updateDevice function