Added 'update' functionality