Improve the latest learning alogirhtm