Updated the CI script