Adicionando imports e corrigindo classes incorretas.