adicionado training set e criando classes para organizar melhor o codigo