code almost finished + *.train + *.btest