Update cs_learn.cpp