Fixed the size issue of data points in pegasos