Made some modifications to COmodel and RLmodel