diff --git a/main.cpp b/main.cpp
index 4edbf27..c8f1a1a 100644
--- a/main.cpp
+++ b/main.cpp
@@ -1,169 +1,73 @@
 #include <iostream>
-#include <fstream>
+#include <vector>
 #include "user.h"
 #include "auth.h"
+#include "db.h"
 
-// --- Tableau d'utilisateurs ---
-User users[] = {
-    {"admin", "1234", ADMIN},
-    {"prof1", "abcd", PROF},
-    {"student1", "pass", STUDENT},
-    {"student2", "123", STUDENT},
-    {"student3", "abc", STUDENT}
-};
-int nbUsers = 5;
+int main() {
+    std::vector<Student> students = {
+        {"student1", {15,12,18}},
+        {"student2", {14,10,16}},
+        {"student3", {13,11,17}}
+    };
 
-// --- Tableau d'étudiants et notes ---
-struct Student {
-    std::string nom;
-    int notes[3];
-};
+    std::vector<User> users = {
+        {"admin","1234",ADMIN},
+        {"prof1","abcd",PROF},
+        {"student1","pass",STUDENT},
+        {"student2","123",STUDENT},
+        {"student3","abc",STUDENT}
+    };
 
-Student etudiants[] = {
-    {"student1", {15, 12, 18}},
-    {"student2", {14, 10, 16}},
-    {"student3", {13, 11, 17}}
-};
-int nbEtudiants = 3;
+    std::string login, password;
+    std::cout << "Login : "; std::cin >> login;
+    std::cout << "Mot de passe : "; std::cin >> password;
 
-// --- Fonctions de vues selon rôle ---
-void viewAdmin() {
-    std::cout << "\n--- VUE ADMIN ---\n";
-    std::cout << "Liste des étudiants et notes :\n";
-    for(int i=0; i<nbEtudiants; i++) {
-        std::cout << etudiants[i].nom << " : ";
-        for(int j=0;j<3;j++) std::cout << etudiants[i].notes[j] << " ";
-        std::cout << "\n";
-    }
-}
-
-void viewProf() {
-    std::cout << "\n--- VUE PROF ---\n";
-    std::cout << "Liste des étudiants et notes :\n";
-    for(int i=0; i<nbEtudiants; i++) {
-        std::cout << i << ". " << etudiants[i].nom << " : ";
-        for(int j=0;j<3;j++) std::cout << etudiants[i].notes[j] << " ";
-        std::cout << "\n";
+    int index = authentifier(login,password,users);
+    if(index == -1){
+        std::cout << "Identifiants incorrects.\n";
+        return 1;
     }
 
-    int id, noteIndex, newNote;
-    std::cout << "Entrez l'ID étudiant à modifier (-1 pour quitter) : ";
-    std::cin >> id;
-    if(id>=0 && id<nbEtudiants){
-        std::cout << "Quelle note modifier (0-2) ? ";
-        std::cin >> noteIndex;
-        std::cout << "Nouvelle note : ";
-        std::cin >> newNote;
-        etudiants[id].notes[noteIndex] = newNote;
-        std::cout << "Note modifiée avec succès !\n";
-    }
-}
+    User currentUser = users[index];
 
-void viewStudent(const User& user) {
-    std::cout << "\n--- VUE STUDENT ---\n";
-    for(int i=0; i<nbEtudiants; i++) {
-        if(etudiants[i].nom == user.login) {
-            std::cout << "Nom : " << etudiants[i].nom << "\nNotes : ";
-            for(int j=0;j<3;j++) std::cout << etudiants[i].notes[j] << " ";
+    // Vues
+    if(currentUser.role == ADMIN){
+        std::cout << "--- ADMIN ---\n";
+        for(auto& s : students){
+            std::cout << s.nom << " : ";
+            for(int n : s.notes) std::cout << n << " ";
             std::cout << "\n";
-            return;
         }
-    }
-    std::cout << "Aucune information trouvée.\n";
-}
-
-// --- Fonctions d'export ---
-void exportAdmin() {
-    std::ofstream file("data.txt");
-    for(int i=0; i<nbEtudiants; i++){
-        file << etudiants[i].nom << " ";
-        for(int j=0;j<3;j++) file << etudiants[i].notes[j] << " ";
-        file << "\n";
-    }
-    file.close();
-    std::cout << "Export complet réalisé (ADMIN).\n";
-}
-
-void exportNotes(const User& user) {
-    std::ofstream file("data.txt");
-    for(int i=0; i<nbEtudiants; i++){
-        if(user.role == STUDENT && etudiants[i].nom != user.login) continue;
-        file << etudiants[i].nom << " ";
-        for(int j=0;j<3;j++) file << etudiants[i].notes[j] << " ";
-        file << "\n";
-    }
-    file.close();
-    std::cout << "Export réalisé.\n";
-}
-
-// --- Fonctions d'import ---
-void importAdmin() {
-    std::ifstream file("data.txt");
-    std::string nom;
-    int n1,n2,n3;
-    int i=0;
-    while(file >> nom >> n1 >> n2 >> n3 && i<nbEtudiants){
-        etudiants[i].nom = nom;
-        etudiants[i].notes[0] = n1;
-        etudiants[i].notes[1] = n2;
-        etudiants[i].notes[2] = n3;
-        i++;
-    }
-    file.close();
-    std::cout << "Import complet réalisé (ADMIN).\n";
-}
-
-void importProf() {
-    std::ifstream file("data.txt");
-    std::string nom;
-    int n1,n2,n3;
-    while(file >> nom >> n1 >> n2 >> n3){
-        for(int i=0;i<nbEtudiants;i++){
-            if(etudiants[i].nom == nom){
-                etudiants[i].notes[0]=n1;
-                etudiants[i].notes[1]=n2;
-                etudiants[i].notes[2]=n3;
+    } else if(currentUser.role == PROF){
+        std::cout << "--- PROF ---\n";
+        for(int i=0;i<students.size();i++){
+            std::cout << i << ". " << students[i].nom << " : ";
+            for(int n : students[i].notes) std::cout << n << " ";
+            std::cout << "\n";
+        }
+        int id, noteIndex, newNote;
+        std::cout << "ID étudiant à modifier (-1 pour quitter) : ";
+        std::cin >> id;
+        if(id>=0){
+            std::cout << "Note (0-2) : "; std::cin >> noteIndex;
+            std::cout << "Nouvelle note : "; std::cin >> newNote;
+            students[id].notes[noteIndex] = newNote;
+        }
+    } else { // STUDENT
+        std::cout << "--- STUDENT ---\n";
+        for(auto& s : students){
+            if(s.nom == currentUser.login){
+                std::cout << s.nom << " : ";
+                for(int n : s.notes) std::cout << n << " ";
+                std::cout << "\n";
             }
         }
     }
-    file.close();
-    std::cout << "Import des notes réalisé (PROF).\n";
-}
 
-// --- MAIN ---
-int main() {
-    std::string login, password;
-
-    std::cout << "Login : ";
-    std::cin >> login;
-
-    std::cout << "Mot de passe : ";
-    std::cin >> password;
-
-    int index = authentifier(login, password);
-
-    if(index != -1) {
-        std::cout << "Connexion réussie !" << std::endl;
-
-        switch(users[index].role) {
-            case ADMIN:
-                viewAdmin();
-                exportAdmin();
-                importAdmin();
-                break;
-            case PROF:
-                viewProf();
-                exportNotes(users[index]);
-                importProf();
-                break;
-            case STUDENT:
-                viewStudent(users[index]);
-                exportNotes(users[index]);
-                break;
-        }
-    } else {
-        std::cout << "Identifiants incorrects." << std::endl;
-    }
+    // Export / Import
+    exportData("data.txt", students);
+    importData("data.txt", students);
 
     return 0;
 }
\ No newline at end of file