Improve CI config