Expanded the trainngset