Update CI code