Update COLA

Changed cola calc logic. Used the function from JSFiddle