Add CI finding code and altered plotting code