Created bigger test and training sets by modifying script