Sort out algorithm and test program
More training data